pay/views.py

@@ -331,12 +331,14 @@ def wx_notify_url_api(request):

                     return HttpResponse(settings.WXPAY_NOTIFY_SUCCESS)
              

                 
              

                 
              

                +@logit
              

                +@transaction.atomic
              

                 def wx_balance_withdraw_api(request):
              

                     user_id = request.POST.get('user_id', '')
              

                 
              

                     # 用户校验
              

                     try:
              

                -        user = UserInfo.objects.get(user_id=user_id)
              

                +        user = UserInfo.objects.select_for_update().get(user_id=user_id)
              

                     except UserInfo.DoesNotExist:
              

                         return response(UserStatusCode.USER_NOT_FOUND)
              

                 
              
@@ -358,9 +360,9 @@ def wx_balance_withdraw_api(request):

                     wxpay = WeChatPay(wechat.get('appID'), wechat.get('apiKey'), wechat.get('mchID'), mch_cert=wechat.get('mch_cert'), mch_key=wechat.get('mch_key'))
              

                 
              

                     if withdraw_type == 'TRANSFER':
              

                -        wxpay.transfer.transfer(user.wx_uid, amount, u'摄影师余额提现，企业付款', check_name='NO_CHECK')
              

                +        ret_data = wxpay.transfer.transfer(user.wx_uid, amount, u'摄影师余额提现，企业付款', check_name='NO_CHECK')
              

                     elif withdraw_type == 'PACKET':
              

                -        wxpay.redpack.send(
              

                +        ret_data = wxpay.redpack.send(
              

                             user.wx_uid,
              

                             amount,
              

                             send_name=wechat.get('redpacket', {}).get('SEND_NAME'),
              
@@ -370,4 +372,8 @@ def wx_balance_withdraw_api(request):

                             remark=wechat.get('redpacket', {}).get('REMARK'),
              

                         )
              

                 
              

                +    # 根据 ret_data 判断是否提现成功, 成功则减余额, 失败则提示
              

                +    user.balance -= amount
              

                +    user.save()
              

                +
              

                     return response(200, 'Withdraw Success', u'提现成功', {})